from models import MarginNet